l=[[1,2,3],[4,5,6],[7,8,9]]
b=[l[i][i] for i in range(len(l))]
print(b)